package 剑指Offer.中等;

import java.util.LinkedHashMap;
import java.util.Map;

/**
 * 请实现 copyRandomList 函数，复制一个复杂链表。在复杂链表中，每个节点除了有一个 next 指针指向下一个节点，还有一个 random 指针指向链表中的任意节点或者 null。
 * <p>
 *  
 * <p>
 * 示例 1：
 * <p>
 * <p>
 * <p>
 * 输入：head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
 * 输出：[[7,null],[13,0],[11,4],[10,2],[1,0]]
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/fu-za-lian-biao-de-fu-zhi-lcof
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 * @author caihe
 * @date 2022/9/26 9:13
 */
public class _35复杂链表的复制 {


    public Node copyRandomList(Node head) {
        if (head == null) {
            return null;
        }
        // 使用一个map进行维护和保存
        // 先保存next主链表
        Map<Node, Node> map = new LinkedHashMap<>();
        Node cur = head;
        while (cur != null) {
            map.put(cur, new Node(cur.val));
            cur = cur.next;
        }
        cur = head;
        while (cur != null) {
            // 获取新节点
            Node nodeCur = map.get(cur);
            // 设置新节点的next和random
            nodeCur.next = map.get(cur.next);
            nodeCur.random = map.get(cur.random);
            // 维护下一个
            cur = cur.next;
        }
        // 返回新链表的头节点
        return map.get(head);
    }

}

class Node {
    int val;
    Node next;
    Node random;

    public Node(int val) {
        this.val = val;
        this.next = null;
        this.random = null;
    }
}
